package com.zixuan.sina.utils;

import cn.hutool.core.io.FileUtil;
import com.aliyun.oss.ClientException;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.OSSException;
import com.aliyun.oss.common.auth.CredentialsProviderFactory;
import com.aliyun.oss.common.auth.EnvironmentVariableCredentialsProvider;
import com.aliyun.oss.internal.Mimetypes;
import com.aliyun.oss.model.*;
import com.zixuan.sina.thread.MyUploadThread;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.concurrent.*;

/**
 * 文件上传工具类 用于上传视频/图片信息
 */
@Component
public class FileUploadUtils {
    @Resource
    private MyUploadThread myUploadThread;
    @Resource
    private FutureTask<String> futureTask;
    @Resource
    private FtpUtils ftpUtils;
    /**
     * 普通上传
     *
     * @param file
     * @return
     */
    //todo:接入阿里云上传
    public String upload(MultipartFile file) {


        String endpoint = OssConstant.ADDRESS;
        endpoint = "https://" + endpoint;
        // 阿里云账号AccessKey拥有所有API的访问权限，
        String accessKeyId = OssConstant.ACCESS_KEYID;
        String accessKeySecret = OssConstant.ACCESS_KEYSECRET;
        String bucketName = OssConstant.BUCKET_NAME;
        try {
            // 创建OSSClient实例。
            OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
            //获取上传的文件名称
            String fileName = UUID.randomUUID().toString().replace("-", "") + file.getOriginalFilename();
            InputStream inputStream = file.getInputStream();
            Date date = new Date();
            SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
            String dateText = format.format(date);
            fileName = dateText + "/" + fileName;

            ossClient.putObject(bucketName, fileName, inputStream);
            // 关闭OSSClient。
            ossClient.shutdown();
            //拼接图片地址
//            https://zixuan-online.oss-cn-beijing.aliyuncs.com/8542953.jpg
//            https://zixuan-online.oss-cn-beijing.aliyuncs.com8542953.jpg
            String url = "https://" + bucketName + "." + OssConstant.ADDRESS + "/" + fileName;
            return url;
        } catch (IOException e) {
            e.printStackTrace();
            return e.getMessage();
        }

    }

    /**
     * 分片上传
     */
    public String chunkUpload() throws com.aliyuncs.exceptions.ClientException, ExecutionException, InterruptedException {

        futureTask=new FutureTask<>(myUploadThread);
        Thread thread = new Thread(futureTask);
        thread.start();
        String url = futureTask.get();
        return url;
    }
//    public String chunkUpload(MultipartFile file) throws com.aliyuncs.exceptions.ClientException {
//        // Endpoint以华东1（杭州）为例，其它Region请按实际情况填写。
//        String endpoint = "https://oss-cn-beijing.aliyuncs.com";
//        String url = null;
//        // 从环境变量中获取访问凭证。运行本代码示例之前，请确保已设置环境变量OSS_ACCESS_KEY_ID和OSS_ACCESS_KEY_SECRET。
//        EnvironmentVariableCredentialsProvider credentialsProvider = CredentialsProviderFactory.newEnvironmentVariableCredentialsProvider();
//
//        // 填写Bucket名称，例如examplebucket。
//        String bucketName = "zixuan-online";
//        // 填写Object完整路径，例如exampledir/exampleobject.txt。Object完整路径中不能包含Bucket名称。
//        Date date = new Date();
//        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
//        String dateText = format.format(date);
//        String fileName = UUID.randomUUID().toString().replace("-", "") + file.getOriginalFilename();
//
//        String objectName = dateText + "/" + fileName;
//
//        CountDownLatch latch = new CountDownLatch((int) file.getSize());
//
//
//
//        // 创建OSSClient实例。
//        OSS ossClient = new OSSClientBuilder().build(endpoint, OssConstant.ACCESS_KEYID, OssConstant.ACCESS_KEYSECRET);
//
//        try {
//            // 创建InitiateMultipartUploadRequest对象。
//            InitiateMultipartUploadRequest request = new InitiateMultipartUploadRequest(bucketName, objectName);
//            // 如果需要在初始化分片时设置请求头，请参考以下示例代码。
//            ObjectMetadata metadata = new ObjectMetadata();
//
//            // 初始化分片。
//            InitiateMultipartUploadResult upresult = ossClient.initiateMultipartUpload(request);
//            // 返回uploadId。
//            String uploadId = upresult.getUploadId();
//            // partETags是PartETag的集合。PartETag由分片的ETag和分片号组成。
//            List<PartETag> partETags = new ArrayList<PartETag>();
//            // 每个分片的大小，用于计算文件有多少个分片。单位为字节。
//            final long partSize = (1024 * 1024L) / 2;   //0.5 MB。
//
//            // 根据上传的数据大小计算分片数。以本地文件为例，说明如何通过File.length()获取上传数据的大小。
//            long fileLength = file.getSize();
//            int partCount = (int) (fileLength / partSize);
//            if (fileLength % partSize != 0) {
//                partCount++;
//            }
//            // 遍历分片上传。
//
//            for (int i = 0; i < partCount; i++) {
//                long startPos = i * partSize;
//                long curPartSize = (i + 1 == partCount) ? (fileLength - startPos) : partSize;
//                UploadPartRequest uploadPartRequest = new UploadPartRequest();
//                uploadPartRequest.setBucketName(bucketName);
//                uploadPartRequest.setKey(objectName);
//                uploadPartRequest.setUploadId(uploadId);
//                // 设置上传的分片流。
//                // 以本地文件为例说明如何创建FIleInputstream，并通过InputStream.skip()方法跳过指定数据。
//                InputStream instream = file.getInputStream();
//
//
//                instream.skip(startPos);
//                uploadPartRequest.setInputStream(instream);
//                // 设置分片大小。除了最后一个分片没有大小限制，其他的分片最小为100 KB。
//                uploadPartRequest.setPartSize(curPartSize);
//                // 设置分片号。每一个上传的分片都有一个分片号，取值范围是1~10000，如果超出此范围，OSS将返回InvalidArgument错误码。
//                uploadPartRequest.setPartNumber(i + 1);
//                // 每个分片不需要按顺序上传，甚至可以在不同客户端上传，OSS会按照分片号排序组成完整的文件。
//                //创建线程池任务上传文件
//
//                UploadPartResult uploadPartResult = ossClient.uploadPart(uploadPartRequest);
//                // 每次上传分片之后，OSS的返回结果包含PartETag。PartETag将被保存在partETags中。
//                partETags.add(uploadPartResult.getPartETag());
//            }
//
//
//            // 创建CompleteMultipartUploadRequest对象。
//            // 在执行完成分片上传操作时，需要提供所有有效的partETags。OSS收到提交的partETags后，会逐一验证每个分片的有效性。当所有的数据分片验证通过后，OSS将把这些分片组合成一个完整的文件。
//            CompleteMultipartUploadRequest completeMultipartUploadRequest =
//                    new CompleteMultipartUploadRequest(bucketName, objectName, uploadId, partETags);
//            // 完成分片上传。
//            CompleteMultipartUploadResult completeMultipartUploadResult = ossClient.completeMultipartUpload(completeMultipartUploadRequest);
//            System.out.println(completeMultipartUploadResult.getETag());
//            //https://zixuan-online.oss-cn-beijing.aliyuncs.com/aaa/%E6%96%97%E7%A0%B4%E8%8B%8D%E7%A9%B9%E9%A2%84%E5%91%8A.mp4
//            url = "https://" + bucketName + ".oss-cn-beijing.aliyuncs.com/" + objectName;
//            System.err.println("返回的url地址连接:" + url);
//        } catch (OSSException oe) {
//
//            System.out.println("Error Message:" + oe.getErrorMessage());
//            System.out.println("Error Code:" + oe.getErrorCode());
//            System.out.println("Request ID:" + oe.getRequestId());
//            System.out.println("Host ID:" + oe.getHostId());
//        } catch (ClientException ce) {
//
//            System.out.println("Error Message:" + ce.getMessage());
//        } catch (FileNotFoundException e) {
//            throw new RuntimeException(e);
//        } catch (IOException e) {
//            throw new RuntimeException(e);
//        } finally {
//            if (ossClient != null) {
//                ossClient.shutdown();
//            }
//        }
//
//        return url;
//    }

    /**
     * 多线程分片上传
     */
    public String threadChunkUpload(MultipartFile file) throws com.aliyuncs.exceptions.ClientException {
        String endpoint = "https://oss-cn-beijing.aliyuncs.com";
        String bucketName = "zixuan-online";
        Date date = new Date();
        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
        String dateText = format.format(date);
        String fileName = UUID.randomUUID().toString().replace("-", "") + file.getOriginalFilename();

        String objectName = dateText + "/" + fileName;
        // 创建OSSClient实例
        OSS ossClient = new OSSClientBuilder().build(endpoint, OssConstant.ACCESS_KEYID, OssConstant.ACCESS_KEYSECRET);

        try {
            // 创建InitiateMultipartUploadRequest对象。
            InitiateMultipartUploadRequest request = new InitiateMultipartUploadRequest(bucketName, objectName);
            // 初始化分片。
            InitiateMultipartUploadResult upresult = ossClient.initiateMultipartUpload(request);
            // 返回uploadId。
            String uploadId = upresult.getUploadId();
            // 如果需要在初始化分片时设置请求头，请参考以下示例代码。
            ObjectMetadata metadata = new ObjectMetadata();

            // partETags是PartETag的集合。PartETag由分片的ETag和分片号组成。
            List<PartETag> partETags = new ArrayList<PartETag>();
            // 每个分片的大小，用于计算文件有多少个分片。单位为字节。
//            final long partSize = (1024 * 1024L) / 2;   //0.5 MB。
            final long partSize = file.getSize();   //0.5 MB。

            // 根据上传的数据大小计算分片数。以本地文件为例，说明如何通过File.length()获取上传数据的大小。
            long fileLength = file.getSize();
            int partCount = (int) (fileLength / partSize);
            if (fileLength % partSize != 0) {
                partCount++;
            }
            // 创建线程池
            ExecutorService executorService = Executors.newFixedThreadPool(partCount);
            // 遍历分片上传。
            int i = 0;
//            for (int i = 0; i < partCount; i++) {
            long startPos = i * partSize;
            long curPartSize = (i + 1 == partCount) ? (fileLength - startPos) : partSize;
            UploadPartRequest uploadPartRequest = new UploadPartRequest();
            uploadPartRequest.setBucketName(bucketName);
            uploadPartRequest.setKey(objectName);
            uploadPartRequest.setUploadId(uploadId);
            // 设置上传的分片流。
            // 以本地文件为例说明如何创建FIleInputstream，并通过InputStream.skip()方法跳过指定数据。
            InputStream instream = file.getInputStream();


            instream.skip(startPos);
            uploadPartRequest.setInputStream(instream);
            // 设置分片大小。除了最后一个分片没有大小限制，其他的分片最小为100 KB。
            uploadPartRequest.setPartSize(curPartSize);
            // 设置分片号。每一个上传的分片都有一个分片号，取值范围是1~10000，如果超出此范围，OSS将返回InvalidArgument错误码。
            uploadPartRequest.setPartNumber(i + 1);
            // 每个分片不需要按顺序上传，甚至可以在不同客户端上传，OSS会按照分片号排序组成完整的文件。
            //创建线程池任务上传文件

            UploadPartResult uploadPartResult = ossClient.uploadPart(uploadPartRequest);
            // 每次上传分片之后，OSS的返回结果包含PartETag。PartETag将被保存在partETags中。
            partETags.add(uploadPartResult.getPartETag());
            executorService.execute(new UploadPartTask(ossClient, uploadPartRequest, partETags));
//            }

            // 关闭线程池
            executorService.shutdown();

            // 等待所有分片上传任务完成
            while (!executorService.isTerminated()) {
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }

            // 完成分片上传
            CompleteMultipartUploadRequest completeRequest = new CompleteMultipartUploadRequest(bucketName, objectName, uploadId, partETags);
            CompleteMultipartUploadResult completeResult = ossClient.completeMultipartUpload(completeRequest);

            // 打印上传结果
            System.out.println("文件上传成功，文件URL：" + completeResult.getLocation());

            // 关闭OSSClient
            ossClient.shutdown();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        return null;
    }

    /**
     * ssh ftp上传到远程服务器
     */
    public String sshFtp(MultipartFile file){
        String url=null;
        try {
            ftpUtils.sshSftp( file.getOriginalFilename(), file.getBytes());
            String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));
            String fileName = UUID.randomUUID().toString().replace("-", "")+suffix;
            url=FtpConstant.DNS_NAME+fileName;
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return url;
    }
}
